import random

import pygame

# 屏幕大小的常量
SCREEN_RECT = pygame.Rect(0, 0, 480, 700)
# 刷新的帧率
FRAME_PER_SEC = 60
# 敌机的定时器常量
CREATE_ENEMY_USEREVENT = pygame.USEREVENT
# 英雄发射子弹时间(pygame.USEREVENT是一个常数)
HERO_FIRE_EVENT = pygame.USEREVENT + 1


class GameSprite(pygame.sprite.Sprite):
    """
    飞机大战游戏精灵
    """

    def __init__(self, image_name, speed=1):
        # 调用父类的初始化方法
        super().__init__()

        # 定义对象的属性
        self.image = pygame.image.load(image_name)
        self.rect = self.image.get_rect()
        self.speed = speed

    def update(self):
        # 在屏幕的垂直方向上移动
        self.rect.y += self.speed


class Brckground(GameSprite):
    """
    游戏背景精灵
    """

    def __init__(self, is_alt=False):
        # 调用父类实现精灵根的创建
        super().__init__("./image/background.png")

        # 判断是否交替图像, 如果是, 需要设置初始位置
        if is_alt:
            self.rect.y = -self.rect.height

    def update(self):
        # 1 调用父类的方法实现
        super().update()

        # 2 判断是否移出屏幕, 如果移出屏幕, 将图像设置到屏幕上方
        if self.rect.y >= SCREEN_RECT.height:
            self.rect.y = -self.rect.height
        pass


class Enemy(GameSprite):
    """
    敌机精灵
    """

    def __init__(self):
        # 调用父类的方法, 创建敌机精灵, 同时指定敌机的图片
        super().__init__("./image/enemy1.png")

        # 指定敌机的初始随机速度
        self.speed = random.randint(1, 3)

        # 指定敌机的初始随机位置
        self.rect.bottom = 0
        max_x = SCREEN_RECT.width - self.rect.width
        self.rect.left = random.randint(0, max_x)

    def update(self):
        # 调用父类的方法, 保持垂直方向的飞行
        super().update()
        # 判断是否飞出屏幕, 删除精灵组
        if self.rect.y >= SCREEN_RECT.height:
            # print("飞出屏幕, 需要从精灵组删除")
            self.kill()

        pass

    def __del__(self):
        # 调用kill方法会调用此方法
        # print("敌机挂了 %s" % self.rect)
        pass


class Hero(GameSprite):
    """
    英雄精灵
    """

    def __init__(self):
        super().__init__("./image/me1.png")
        self.speed = 0
        # 设置英雄的初始位置
        self.rect.centerx = SCREEN_RECT.centerx
        self.rect.bottom = SCREEN_RECT.bottom - 120
        # 创建子弹精灵组
        self.buttlt_group = pygame.sprite.Group()

    def update(self):
        # 英雄在水平方向移动
        self.rect.x += self.speed
        # 控制英雄不能离开屏幕
        if self.rect.x <= 0:
            self.rect.x = 0
        elif self.rect.right >= SCREEN_RECT.right:
            self.rect.right = SCREEN_RECT.right

    def fire(self):
        print("fire.....")
        # 创建子弹, 并且加入子弹精灵组
        for i in (0, 1, 2):
            bullet = Bullet()
            self.buttlt_group.add(bullet)
            # 设置子弹的初始位置
            bullet.rect.bottom = self.rect.top - i * 10
            bullet.rect.centerx = self.rect.centerx


class Bullet(GameSprite):
    def __init__(self):
        super().__init__("./image/bullet1.png")

    def update(self):
        self.speed = -1
        super().update()

        # 判断子弹是否飞出屏幕
        if self.rect.bottom <= 0:
            self.kill()

    def __del__(self):
        # print("子弹杯删除")
        pass
